/*
 * @lc app=leetcode.cn id=114 lang=typescript
 *
 * [114] 二叉树展开为链表
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

/**
 Do not return anything, modify root in-place instead.
 */

//  思路：迭代
//  先拉平左右，再接上
//  参考：https://labuladong.github.io/algo/2/19/22/

//  复杂度：O(n)  O(1)

function flatten(root: TreeNode | null): void {
    // base
    if (!root) return

    flatten(root.left)
    flatten(root.right)

    // 1、左右子树已经被拉平成一条链表
    const left = root.left, right = root.right
    // 2、将左子树作为右子树
    root.left = null
    root.right = left
    // 3、将原先的右子树接到当前右子树的末端
    let p = root
    while (p.right) {
        p = p.right
    }
    p.right = right
};

// @lc code=end

//  思路：
//  先序遍历后再进行组合，还可以利用栈，将treeNode送入栈

//  复杂度：O(n)  O(1)
//  通过寻找先驱节点可以将空间复杂度降为O(1)
//  过程是：按先序遍历，当前左子树不为空，则找到左子树最右边的节点，将右子树接上，再将左子树接到右边，循环。

function flatten2(root: TreeNode | null): void {
    const res: any = []
    const firstOrder = (root: TreeNode | null) => {
        if (!root) return
        res.push(root)
        if (root.left) firstOrder(root.left)
        if (root.right) firstOrder(root.right)
    }
    firstOrder(root)

    for (let i = 1; i < res.length; i++) {
        const pre = res[i - 1], curr = res[i]
        pre.left = null
        pre.right = curr
    }
};

import { TreeNode } from './type'

const root = new TreeNode(1, new TreeNode(2, new TreeNode(3), new TreeNode(4)), new TreeNode(5, null, new TreeNode(6)))
flatten(root)